#include "LayerSegInference.h"

int main()
{
	std::string image_name = "test.png";
	std::string image_path = "./test_onnx/input/" + image_name;
	cv::Mat image = cv::imread(image_path, cv::IMREAD_GRAYSCALE);
	cv::Mat outputImg;
	OrtLayerSeg* LayerSeg = new OrtLayerSeg;

	std::vector<int> layerSuface;
	char* ret = LayerSeg->RunSession(image, layerSuface, outputImg);
	cv::imwrite("./output/" + image_name, outputImg);

	return 0;
}
